tensor([[[0.3200, 0.3200],
         [0.3200, 0.3200]],

        [[0.3200, 0.3200],
         [0.3200, 0.3200]],

        [[0.3600, 0.3600],
         [0.3600, 0.3600]]], grad_fn=<RoundBackward1>)